bug fixes & UI