Modify algorithm to display win