Add comments on returns