Minor cleanup
Add UI class
Add player type messages to ui class