Added Unittest class