flax

d75c2e58 - Merge pull request #3588 from canyon289:doc_improvements