Updates to the readme and code comments