adds more unit tests to both models