Solving assignment problem - didn't saved map() implementation