After creating infogain diagrams