Adding command line parameter handling