Adding UI functions in separate class for easy refactoring.