Slight UI improvement