Added missing argument to get_layer_outputs function.