update documentation for flatten

Update documentation for flatten to clarify behavior.