Removed reshaping from ex2_reg.