[IMP] Project : small improvement in code