Assignment 5: UI Enhancements