Cleaned the code a bit...

Signed-off-by: Pier-Luc Plante <pier-luc.plante.1@ulaval.ca>