working on making a better ui