Huge UI revamp

Mostly work from #125
Fix #5