Unit tests

Change-Id: Ib70a34f2ab78277a932acd1ced3ca4eef4140482