diff --git a/+PlotID/removePltIdFiles.m b/+PlotID/removePltIdFiles.m
index 37ef1827ad2049bd261780e4d190f05907a7b676..57a07b7c9a7ea20f7d4a209402729bc469aac232 100644
--- a/+PlotID/removePltIdFiles.m
+++ b/+PlotID/removePltIdFiles.m
@@ -1,19 +1,27 @@
 function [fListClean] = removePltIdFiles(fList)
 %removePltIdFiles removes functions that are part of PlotID out of flist
 %   Detailed explanation goes here
-%addpath('..\fcn_core');
 
 [~,names,ext] = fileparts(fList);
 names = strcat(names, ext); % add ext for comparison
 
 % Get a list of all .m files that are part of Plot id
-PltID_flist = struct2table(dir('+PlotID')); %get list of files 
-[~,~,PltID_flist.ext(:)] = fileparts(PltID_flist.name(:)); % add ext column
-
-PltID_flist = PltID_flist(strcmp(PltID_flist.ext,'.m'),:);
+packageContent = what('PlotID');
+% packageContent.classes has no extensions
+PltID_classlist = packageContent.classes;
 
+% Class Methods need to be listed in an additional function
+Class_flist = cell(1,numel(packageContent.classes)); %preallocate 
+for i=1:numel(packageContent.classes)
+    temp = what(['PlotID',filesep,'@',PltID_classlist{i}]);
+    Class_flist{i} = temp.m; 
+end
+ 
+Class_flist = vertcat(Class_flist{:});
+% all plotID .m files:
+PltID_flist = [packageContent.m; Class_flist]; 
 % Comparison and filter
-fListClean = fList(~ismember(names,PltID_flist.name));
-
+fListClean = fList(~ismember(names,PltID_flist));
 end
 
+