working on matplotlib stuff