Make sure your dog knows they're loved and wanted. The calmness that helps provide helps a lot when you're training.
Also please socialize your dog. Dog parks can be very scary for a puppy, so it's important to try to ease into the socializing. Maybe by other puppy meet ups or if you have friends with dogs.