Update mlp_test.py