Cool! If you are interested, we have open sourced our code: https://github.com/emmyqin/iw_sft