Update example for broadcasting and add elementwise explanation