Adding a rotation function + a prototype in main.c