Added model test, made all the red green