More documentation and small fixes