cleaned up code and output, added another input file and included it in tests.txt