Python matplotlib: first version which works