updated comments, changed for loop

for loop is now a list comprehension