Fixed plotting matplotlib