Added UI logic