Add two buttons function to replace the menu function.