Changes return
Changes model logic