Create build_model.py