It might be nice to remove the use of spark's wrapper class since we are handling serialization on our own now.