Updated documentation. Small code changes.