3 files changed, 9 insertions, 4 deletions
diff --git a/frontend/src/app/_data/Dataset.ts b/frontend/src/app/_data/Dataset.ts
index b4e46315..aaee50eb 100644
--- a/frontend/src/app/_data/Dataset.ts
+++ b/frontend/src/app/_data/Dataset.ts
@@ -5,6 +5,9 @@ export default class Dataset {
         public header: string[] = [],
         public fileId?: number,
         public extension: string = '.csv',
-        public dateCreated: Date = new Date()
+        public isPublic: boolean = false,
+        public accessibleByLink: boolean = false,
+        public dateCreated: Date = new Date(),
+        public lastUpdated: Date = new Date()
     ) { }
 }
 \ No newline at end of file
diff --git a/frontend/src/app/_data/Model.ts b/frontend/src/app/_data/Model.ts
index 162d5d78..1a120ca7 100644
--- a/frontend/src/app/_data/Model.ts
+++ b/frontend/src/app/_data/Model.ts
@@ -3,9 +3,10 @@ export default class Model {
         public name: string = 'Novi model',
         public description: string = '',
         public dateCreated: Date = new Date(),
+        public lastUpdated: Date = new Date(),
         public datasetId?: number,
 
-        //Test set settings
+        // Test set settings
         public inputColumns: number[] = [0],
         public columnToPredict: number = 1,
         public randomOrder: boolean = true,
diff --git a/frontend/src/app/_data/Predictor.ts b/frontend/src/app/_data/Predictor.ts
index 379526ec..77d9498b 100644
--- a/frontend/src/app/_data/Predictor.ts
+++ b/frontend/src/app/_data/Predictor.ts
@@ -2,8 +2,9 @@ export default class Dataset {
     constructor(
         public name: string = 'Novi izvor podataka',
         public description: string = '',
-        public inputs: string[] = [],
-        public output: string = '',
+        public columns: string[] = [],
+        public isPublic: boolean = false,
+        public accessibleByLink: boolean = false,
         public dateCreated: Date = new Date()
     ) { }
 }
 \ No newline at end of file