Add assign_attributes