Small addition to simulation image generator