Update script to include docker