Solution to Programming Assignment Week 2
First submission