Unit Step Function In Python