Add Kryo compatibility