Add more classes to kryo registrator