Fix more documentation