Added more to the Square diamond algo