Added what UI stuff I can, left some stuff out

Wasn't able to test -