more work on week 2 assignment