Add partly working support for Reshape