Move tests into their own module