Adding reshape() function