Changed the test data to be clearer, added new test with inputs