Fixing main function

Signed-off-by: Yann Morère <yann.morere@univ-lorraine.fr>