Finish putting all decorators in mlp.py