improving aesthetics of fig
minor edits to code
minor edits to code
diversity stats & figures including WSprop