add the track for function from return statement