Add more methods to the Diagram view.
Remove useless diagram part.