fixed .map() function