Refactored and changed gauntlet-4 problem