Small styling fixes for the editor