Changed Main example to be IntelliJ friendly