Adding command line input