put it all in a list comp