Fix more docs in math