Wired up UI a bit