Update squared.py