Tests

Added tests for models. 
No need to put this two inside a block.