updates to square_geom
change highlighting to match pycharm

at least, to match it more than currently