Map function improvment
Week HW 3 added