Added in more stuff to ui to make it run better