add rspec and edit that gerry gave me on my code