Week 4, Challenge 5: Update code

Update code to work with rspec