Added way to run main method from gradle. Replaced ints with enums. Added some JUnit Tests