Add collimated plume testing to infotaxis script