Updated to include the Square model.