Modified the script for ml2