Added ability to set model to load from command line.