re-factoring model

Signed-off-by: loopasam <samuel.croset@gmail.com>