Add model serialization capability