Tests

Added tests for models. 
Use the expect syntax. :)