Added flow control challenge code