Fixed a small mistake and added some more documentation