Improved UI boxes design