changed up the UI a bit