First attempt at the assignment, unsure about wording of last question