Fixed bugs involving copy paste code for map2