ENH: Added scripts to replicate figures from amanuscript