Updated docs according to changes in #138

Updated docs to reflect changes in #138