Updated control flow lesson