Input file works in the main code