keeping keys in RDDs in nmf - fixes
fixed the nmf with keeping track of keys