Minor tinkering with plotting