Add unit test and e2e test stuff