Expanded square_root_oracle.py