Add end-to-end tests