Your correct in your reasoning.

Keep in mind at four months you are dealing with a 3 year old in human terms. When training and the dog is distracted you need to get its attention using a toy or some noise maker, a clicker works great for this and can be used to teach it basic commands.

JG