Add support for watermark in Graph class