Small styling bug fixes