Unit testing + add model behaviours