Enhance menu number function