Small styling fixes