ui enhancement

I made a small but really nice ui enhancement which needs to be reviewd