Added component diagram correction