Update q4_info_gain.m