More UI + bug fix

Added a bit more UI and fixed a bug I was having with painting on
canvas.