Homework solution for lesson 2 of ruby