Charlie and the Chocolate Factory sequel

I’ve often thought of a sequel story to Rahl Dahl’s book. There would be the story of Charlie Bucket trying to run the factory and find happiness there as well. How about if Charlie stumbles upon a treasure hunt of sorts that Willie left for him? Or what if the factory becomes “infected” by the spirit of the naughty children and Charlie and Willie have to face them one more time? What if Charlie loses the title to the factory from some legal paperwork and has to fight an antagonist to win it back?