Updated test scripts for ml stuff