Adding e2e support