Added swarm plugin and refactored puppet setup