Add command line argument handling