UI updates. Menu fixes