Little UI imrpovement.