Unit tests

Change-Id: Ib47e178f19f45d40d541f2e39d6048adaa86e824