Kinda want to work on the UI a little bit