list comp

change some for loop to list comp