Simply changed exercise to match all cases