Adding IntelliJ IDEA support