made changes to make it work with intellij and maven