changed map2 a bit