Step Function In Python