And preferably use fileinput for the stdin so that you can name files on the command line as well