Jag använder jag en switch / behållarkombination?

Här är min nuvarande situation. Jag har 14/2 till taket. Den heta kabeln har en löpare som går ner till strömbrytaren och den andra ledningen är omkopplad varm, tillbaka till ljuset. Den andra sidan av ljuset är ansluten till 14/2 tillbaka till brytaren.

Något geni i det förflutna, bytte ljusströmbrytaren med en kontakt för omkopplare. Istället för att köra om 14/3 gick de helt enkelt varmt till behållaren och hoppade den neutrala sidan rakt mot marken. Jag har kopplat bort detta, men har nu en icke-funktionell behållare där jag verkligen behöver en.

Jag vet att jag behöver köra en 14/3 ner till strömbrytaren, och kan enkelt göra detta arbete på ett sätt som är säkert och funktionellt. Men jag skulle också vilja att det var korrekt, så att ingen framtida person löper några överraskningar.

Så, vad är lämpliga färger att använda i min B / W / R 14/3?

Jag tycker att det borde vara en B-B wirenut i taket, kör den svarta till Switch. R Bör vara omkopplad varm till ljuset och ljuset tillbaka till Line Neutral (White).

Nu, för behållaren, kan jag ta linjen från den heta till strömbrytaren och köra den fria vita löparen tillbaka till linjen Neutral.

Är detta rätt? Vad skulle en erfaren elektriker tro om han kom över det i framtiden?

EDIT: I det andra diagrammet är neutralen och marken inte anslutna.

RESOLVED: Måste bryta gipsskivan för att få det gjort vilket ledde till ett patchjobb. Den gamla tråden häftades till studsarna och jag var tvungen att släppa den för att använda den som en fisk. Markerade röda med elektriskt band.