File Classification Fix

Fixed an issue with how models were being classified by extension.