Add necessary returns