Tidy up codebase

[peterbjornx]

The code is currently very sparsely commented and all of the UI logic is in a single source unit. This makes the codebase somewhat unintuitive.