Add homework answer for week 5