Added diagram option