## Refactor code and remove duplicate code ## handle graphs more gracefully (e.g close plts otherwise they'll be stuck in memory)
