modified arrow function visitor to visit the replacement function
re-added the arrow function expression visitor