Make use of the new soft_wraps decorator in mlp.py