Add e2e testing