Matplot Visual Code Python