Change script to run container