WIP: Serialization

This probably will become obsolete, but in this MR I basically solved some serialization issues.

We no longer need to use functools.partial as callables in the legacy classes.