Clean up python