make the diagram much prettier