Tests

Added tests for models. 
```updates the session```