Add end game controller