Update WGIntegration.java

Cleanup wg integration
Merge pull request #5 from Shevchik/patch-4

Update WGIntegration.java