Working on week 5 assignment