Added last tests to input. Have to get a proper txt to test this with