little UI corrections