Incorporate missing features in the new NiW code

[lucasaugustomcc]

Regina wrote as a comment in the paper (v7) that the repository she provided has a few more features than the actual repository.

• What is that repository? (GitHub or Google Drive)?
• What are those features?

[reginawang3495]

GitHub(it's on the comment)
I will compare with the notebook while I comment

[reginawang3495]

This is basically the same question as the first