UI improvements and responsiveness