modify sklearn tutorial on features