Created a UI class for logic testing