made more UI improvements