Updated docs to reflect the latest code