Modification du main

Pour qu'il puisse compiler !