UI Logic moved to its own class