update for challenge problem