UI: Menu functionality work