adding some UI love