Modify menu function