no reason to keep the Code class - now we could store the source for graph in the attributes