Add command line arguments support