add support for hg