How to use matplotlib in python