list comp refactor