Add function to post-process the result