Added IntelliJ support