Final Code (YELLOW)