Updated end to end tests