Unit tests

Change-Id: I94ea83893e10d5f6e9d394d68ab89b3ba1cd2427