Add end game function