Start on solution for edx ml course hw1 pla problem