Add mother class
Return object instead of array
Add entities model