Added a command line argument (mem-mod) for selecting a memory model.