Changed diagram slightly