Unit tests

Change-Id: Ieab9814d612ade1854de4a52beeb15cef2fcd6c1