Updating to make it work by passing command line args.