This is my first assignment for the data scientist toolbox course on coursera