Added 2 types of files plus the related logic for identifying file types.