Add post-processing steps