The submission for https://class.coursera.org/predmachlearn-004