added some UI love