Optimization of high energy collimator design