Minor code and documentation update.