cleaned up diagram and some text, minor