Cleaned up 'build.py'