Cleaning up python