added udacity lesson 3 problem set