Minor bug fix and updated documentation