Add puppet modules support