minor updates per the assignment