Version 0.1: bug fix in post-treatment