red light, green light codechef solution