Update plot_rfe_digits.py

Import statement have been moved to the beginning of the file.