minor changes. Homework completed! :)