Lists Python Hackerrank Solution