Add e2e testing support