Adding more TDD (Rspec)