Move tests to separate package