Small styling updates