Fixed unit cell code in build.py