Added serialization to each model class