Cleaning up code for push.