Code plus propre mais ne marche pas completement