Update README and the module diagram