Add input file handling