Documentation and a small fix