modified steps py file