Correction du code

Amélioration du code et précision de type plus "précise"