Add some tests for models