Testing out loops and list funtionality with python