Move menu definition to __init__