Added a tests class