added UI elements, prettied it up