Improve serialization

[lext]

Current serialization needs improvement. I does not work correctly all the time. We also need to make sure that we can load the transforms back after serialization