Convert all layers to use initialize