check in week 3 assignment - update with functionality from week 2
check in week 3 assignment, tests pass