Finished adding unit tests for model