Tests

Added tests for models. 
Use expect syntax.