Plot UI improvements