Some small fixes and docs