add kryo serialization