Fixed pyplot imports