Add support for hg