Added first attempt at problem 2