Add jest testing