Add optional method for post processing