Updated image to match code enhancements