Add rspec tests and relationship model