Added postgame logic