Done with unit testing of models