Adding clean up and comments. Winner button functionality