This is the Project 2 from week 2