updated example plots, cleaned up script