Fix error and change documentation