Fix for command line input

STDIN didn't work. This is what fixed it for me.