Updated unit test classes to be public so that they run again