Update rock.py