1 files changed, 4 insertions, 3 deletions
diff --git a/pokemodr/models/AbilityModel.h b/pokemodr/models/AbilityModel.h
index 6221fddf..0a012845 100644
--- a/pokemodr/models/AbilityModel.h
+++ b/pokemodr/models/AbilityModel.h
@@ -24,7 +24,7 @@
 // Forward declarations
 class AbilityEffectModel;
 
-class AbilityModel : public ObjectModel
+class AbilityModel : public GroupObjectModel
 {
     public:
         AbilityModel(BaseModel* parent, Object* object);
@@ -38,14 +38,15 @@ class AbilityModel : public ObjectModel
         BaseModel* childItem(const int row);
         int indexNumber() const;
         
-        bool canInsertRows() const;
         bool insertRows(const int rows);
         
-        bool canRemoveRows() const;
         bool removeRows(const int position, const int rows);
         
 //         bool dropMimeData(const QMimeData* data, Qt::DropAction action, int row, int column);
     private:
+        void setupData();
+        void clearData();
+        
         QList<AbilityEffectModel*> m_effects;
 };