Updated docs, small bug fixed