Updated docs with improvements