Separate UI stuff from game engine to UI Module