Using neural networks to build a better policy