working on ui a bit