Tests

Added tests for models. 
Too much indent.