little UI improvement