add return statements