Added docstest code