1 files changed, 10 insertions, 0 deletions
diff --git a/src/main.c b/src/main.c
index a841322..5db5fb7 100644
--- a/src/main.c
+++ b/src/main.c
@@ -12,6 +12,7 @@
 /* ************************************************************************** */
 
 #include <libft.h>
+#include <minishell.h>
 #include <stdlib.h>
 #include <stddef.h>
 #include <unistd.h>
@@ -25,8 +26,14 @@ int
 	uint8_t i;
 
 	i = 0;
+	arg = NULL;
 	while (1)
 	{
+		if (arg)
+		{
+			free(arg);
+			arg = NULL;
+		}
 		if (!(arg = (char*)ft_calloc(129, sizeof(char))))
 			return (1);
 		write(1, "joe-shell~> ", 12);
@@ -44,7 +51,10 @@ int
 			i++;
 		}
 		arg[i] = '\0';
+		if (ft_process_arg(arg) == 8)
+			break ;
 	}
 	free(arg);
+	arg = NULL;
 	return (0);
 }