diff --git a/main.c b/main.c
index 0872076..3a80ea4 100644
--- a/main.c
+++ b/main.c
@@ -14,14 +14,21 @@ void clean_exit(int error_code){
 	exit(0);
 }
 
-int main() {
+int main(int argc, char* argv[]) {
 	SharedData * shared_data = GET_INSTANCE(SharedData);
 
 	signal(SIGINT, clean_exit);
-	shared_data->set_main_building(shared_data, NEW(Building, "../residents.txt", "../visitors.txt"));
+	if(argc == 3){
+		shared_data->set_main_building(shared_data, NEW(Building, argv[1], argv[2]));
+	} else if (argc == 1){
+		shared_data->set_main_building(shared_data, NEW(Building, "../residents.txt", "../visitors.txt"));
+	} else{
+		CRASH("Arguments invalides\nUsage : ./LO41 [residents_file visitors_file]\n");
+	}
+
 	shared_data->start_all_threads(shared_data);
 	shared_data->wait_all_threads(shared_data);
 	DELETE(shared_data);
 
 	return 0;
-}
\ No newline at end of file
+}