Add keep going functionality