transfer code notebook -> python ; redo cleaner figures (in ggplot for paper uniformity)
transfer code notebook -> python ; redo cleaner figures (in ggplot for paper uniformity)