Add audio control function