Added remove graph function (not implemented yet)