list compression instead of loop - not tested