Add comments and add check to make sure set contains the mid being predicted on