Update script to run end-to-end