1 files changed, 11 insertions, 4 deletions
diff --git a/LoadFilesCommand.h b/LoadFilesCommand.h
index e13d873..19913d8 100644
--- a/LoadFilesCommand.h
+++ b/LoadFilesCommand.h
@@ -5,14 +5,21 @@
 
 #include "Command.h"
 
+class Museum;
+
 class LoadFilesCommand : public Command {
-	using Command::Command;
+public:
+	LoadFilesCommand(Museum & m, std::vector<std::string> files);
+	LoadFilesCommand(Museum & m, int argc, char ** argv);
 
 public:
-	virtual void execute(std::vector<std::string> files);
-	virtual void execute(int argc, char ** argv);
+	virtual void execute();
+
+private:
+	void load_files();
 
 private:
-	void load_files(std::vector<std::string> files);
+	Museum & museum;
+	std::vector<std::string> files;
 };