Implement ui drawing

Don't actually merge this!