Modify the media type classification method