tiny bit of clean up
adding a little ui