Lightweight version of the stray light checking code for NuSTAR