Fixing a bug + documentation