Deserialize correctly

Signed-off-by: Ian Macalinao <ianmacalinao@gmail.com>
Register class with serialization

Signed-off-by: Ian Macalinao <ianmacalinao@gmail.com>