added another test and removed main method