Script for optimising (based on all 5,000 examples)