Add pause / restart / game ended functions