improved ui to look more like current ui