Updated to work with the latest nuget packages and gen2 Lightning.