second pull request

fixed homework based on feedbacks that I got