Combined into one UI class again. Added color coding depending on UI id