Rewrite expand_menu method