Added the class to controlling output directory in unit tests