New tests

tests and stuff for models