Made it so you can specify the input file - doh.