Update rotate.py