Update script to support audio