Code update

Updated code to accept command line arguments