Add more code to menu module.