diff --git a/f_bonus_print.c b/f_bonus_print.c
index 56b9512..f58ee9e 100644
--- a/f_bonus_print.c
+++ b/f_bonus_print.c
@@ -59,7 +59,6 @@ int			print_binary_map(unsigned int *map, int size, int *dope)
 		print_sized_map(map, size, size, '#');
 		ft_putchar('\n');
 	}
-	free(map);
 	return (size);
 }
 
diff --git a/f_print.c b/f_print.c
index d5ab0d2..01c14ee 100644
--- a/f_print.c
+++ b/f_print.c
@@ -1,7 +1,7 @@
 /* ************************************************************************** */
 /*                                                                            */
 /*                                                        :::      ::::::::   */
-/*   print.c                                            :+:      :+:    :+:   */
+/*   f_print.c                                          :+:      :+:    :+:   */
 /*                                                    +:+ +:+         +:+     */
 /*   By: hulamy <hulamy@student.42.fr>              +#+  +:+       +#+        */
 /*                                                +#+#+#+#+#+   +#+           */
@@ -13,8 +13,8 @@
 #include "fillit.h"
 
 /*
-** function that print a map of height and width
-** usefull to print tetris
+** Function that print a map of height and width
+** useful to print tetris
 */
 
 void	print_sized_map(unsigned int *tab, int width, int height, char letter)
@@ -44,7 +44,7 @@ void	print_sized_map(unsigned int *tab, int width, int height, char letter)
 /*
 ** Print the final map with the letters
 ** if flag value is 0 -> print moulinette version
-** if flag value is 0 -> print in color
+** if flag value is p -> print in color
 */
 
 char	*init_print_map(t_fillist *list, int size)
@@ -53,7 +53,8 @@ char	*init_print_map(t_fillist *list, int size)
 	int			i;
 	int			j;
 
-	map = (char *)malloc(sizeof(*map) * (size * size + 1));
+	if (!(map = (char *)malloc(sizeof(*map) * (size * size + 1))))
+		return (NULL);
 	map[size * size] = '\0';
 	i = -1;
 	while (++i < size * size)
diff --git a/main.c b/main.c
index 3161953..781876c 100644
--- a/main.c
+++ b/main.c
@@ -30,7 +30,7 @@ int		*create_dope(char *av, int mdp)
 
 	comp = "dope";
 	if (!(dope = (int*)malloc(sizeof(*dope) * 4)))
-		return (NULL);
+		return (0);
 	i = 0;
 	while (i < 4)
 		dope[i++] = 0;
@@ -110,5 +110,6 @@ int		main(int ac, char **av)
 	else
 		print_error("usage: Please submit a file.\n> ./fillit file.fillit\n");
 	free(dope);
+	while (1) ;
 	return (0);
 }