Add statement before return statement