add ml tests