Test class added to solution