Done write rspec test for model