slight update and moved to using TDD and rspec