Fixing bug with classify feature #61