Garage wiring idea

Currently in a receptacle controlled by the switch. Was thinking if the power is always on, then lights would go in smart switch with the three ways controlling (OK, not really three ways then - just toggles). And as I said, I;m open to any ideas.