Works with command line input now!