Use Kryo for serialization