changed code diagram in readme