some updates on the ui