I did a two step. First, considerable severe short term punishment (they have a short memory). Then, and you can work on the details during that short term, spend bunches of time with him, in lots of small pieces (their attention span is short) and sustain that. That constant communication keeps his head from getting to far away. My almost 30 year old son (I'm getting old) would go through that cycle every couple of months.