Resolve issue #6
minor edits