Add working function body