chanigng device - really need to make this an arg
Change path for Models to current dir - hack for running from source
add index to cell id mapping, its more useful than idToIdx (we can get that from the cell states). idToIdx can be easily removed