Fix some styling issues