Fix up styling