Added e2e testing
Closes #2