Add regression tests