change to StorageLevel.MEMORY_ONLY_SER