Fixes #29

made changes to the references in the code