added code for week4 day2
added code for week4 day2