Arctic Wolf Interview Question

How would I change my code if I knew all requirements from the beginning? (from the live coding session)