Added tests class