Solution fix and added input.txt