Adding more documentation. Also un-commenting critical part.