working on visual improvements