Move tests to a separate class.