Unit tests

Change-Id: I49c16932e685e1e9d824755523a5221455647bc9