Reverting to apply instructor's solution
Updated to reflect challenge one solution