Update entropy example with improved matplotlib usage