added initial code. Passed rspec