Put all UI management methods into a seperate class