UI formatting / code tidy up