Adding plotting functionality to the editor