Modification to find local beam intensity

This image properly outputs and takes about 3s to process on my old laptop