Hackerrank python data science