Data preparation for the Pet dataset