Updated code to work with "Representation"->"Feature" renaming.