Update documentation
updated comments