Slight update to classifier interface