Unit tests

Change-Id: Ia5ea30683763b7b0e4c1c76c992019711a5f0744