Update documentation, small bugfix