Edit controller2 to cope up with dren's graph