Refactor to have internal package and creating TestManager API