Added a puppet control class