Update cell graph

Small changes to the cell graph:

• moving to the same implementation than the task plots
• adding a new argument to swift
• writes the files every n steps and not only at the beginning.

@rgb Here is the merge request with the changes that I mentioned in my email.