Started on adding rspec tests