Documentation update and also removing some print statements