Fix more styling