Homework 2

Made my homework. And sending a pull request