Updated readme and code comments