Committed the code for the reshaping experiment branch.