Unit tests

Change-Id: Ia37d9ec98dd560f482312d73381154f4cc018c0d