nested lists hackerrank solution in python