we got a bit confused with the diagram task