Unit tests

Change-Id: Id775e724559559863f88c8875c6cc608377839ce