Hackerrank Data Science Challenge