Fix up visuals a bit