Update to support the new version of intellij plugin

Closes #6