Clean up python setup