* Update test.sh for new model
