Add post-script functionality