Modified build-models.sh