Add attributes to message model