Correcting small bug
Correcting documentation