Got the first model test to pass