Forgot to output 'out_final' rather than 'out'