Adding tests and rspec