Minor refactoring.
Use rootpy instead of PyROOT directly.