Adding toBeInDOM and tests