Bug Fixes & UI Enhancements