fix for import error in training coding question