Remove awkwardly written reshape