Add tests

Change-Id: I8cc1495298892d67214306ae2ec06ae7b18cb1e6