Working on week4 assignment