--- UserCode/Morgan/interface/LeptonAnalyzer.h	2009/04/17 15:17:38	1.1
+++ UserCode/Morgan/interface/LeptonAnalyzer.h	2009/06/10 11:17:05	1.3
@@ -25,26 +25,32 @@
 #include "FWCore/Framework/interface/EventSetupRecordImplementation.h"
 
 #include "../interface/TRootEvent.h"
+#include "../interface/TRootBeamSpot.h"
+
 
 class LeptonAnalyzer
 {
-	
-public:
-	LeptonAnalyzer(const edm::ParameterSet& producersNames);
-	LeptonAnalyzer(const edm::ParameterSet& producersNames, int verbosity);
-	~LeptonAnalyzer();
-	void SetVerbosity(int verbosity) {verbosity_ = verbosity; };
-	void initIPCalculator(const edm::Event& iEvent, const edm::EventSetup& iSetup, TRootEvent* rootEvent);
-	float IP3DSignificance(const reco::TransientTrack& track);
-
-protected:
-	int verbosity_;
-	std::string dataType_ ;
-	edm::InputTag vertexProducer_;
-	bool dummyVertex_;
-	edm::ESHandle<TransientTrackBuilder> trackBuilder_;
-	const reco::Vertex* primaryVertex_;
-	const GlobalPoint* vertexPoint_;
+
+   public:
+      LeptonAnalyzer(const edm::ParameterSet& producersNames, const edm::ParameterSet& myConfig, int verbosity);
+      ~LeptonAnalyzer();
+      void setVerbosity(int verbosity) {verbosity_ = verbosity; };
+      void initIPCalculator(const edm::Event& iEvent, const edm::EventSetup& iSetup, TRootEvent* rootEvent, TRootBeamSpot* rootBeamSpot);
+      float ip3DSignificance(const reco::TransientTrack& track);
+
+   protected:
+      int verbosity_;
+      bool allowMissingCollection_;
+      bool initOK_;
+      std::string dataType_ ;
+      bool doPrimaryVertex_;
+      bool doBeamSpot_;
+      edm::InputTag vertexProducer_;
+      bool dummyVertex_;
+      edm::ESHandle<TransientTrackBuilder> trackBuilder_;
+      const reco::Vertex* primaryVertex_;
+      const GlobalPoint* vertexPoint_;
+
 };
 
 #endif