Tests

Added tests for models. 
No need to get the parameter. See how you did this in answer factory.