Issue with python lists and loops