Add unit testing utilities
Update documentation