Added the two more tests to project