Add focus menu function