Add IntelliJ support