Added answer for week 4