Update docs and tests for equations