HW5: Added UI