WIP = 2nd programming assignment of machine learning