Update mlp.py