Add support for jest