Unit tests

Change-Id: Idd5eded368b9dcd852fd6e0ad9d429b0a3970c95