change the mistakes

Author: KhapalSofiia

src/main/java/core/basesyntax/Application.java

@@ -1,9 +1,10 @@
 package core.basesyntax;
 
 public class Application {
+    private static final int NUMBER_OF_BALLS = 3;
     public static void main(String[] args) {
         Lottery lottery = new Lottery();
-        Ball[] balls = new Ball[3];
+        Ball[] balls = new Ball[NUMBER_OF_BALLS];
         for (int i = 0; i < 3; i++) {
             balls[i] = lottery.getRandomBall();
             System.out.println(balls[i].toString());

src/main/java/core/basesyntax/Ball.java

@@ -1,16 +1,14 @@
 package core.basesyntax;
 
 public class Ball {
-    private String color;
+    private Color color;
     private int number;
-    public void setColor(String color) {
+    public Ball(Color color, int number){
         this.color = color;
-    }
-    public void setNumber(int number) {
         this.number = number;
     }
     public String getColor() {
-        return color;
+        return color.toString();
     }
     public int getNumber() {
         return number;

src/main/java/core/basesyntax/Color.java

@@ -1,10 +1,10 @@
 package core.basesyntax;
 
 public enum Color {
-    pink,
-    blue,
-    yellow,
-    green,
-    red,
-    purple
+    PINK,
+    BLUE,
+    YELLOW,
+    GREEN,
+    RED,
+    PURPLE
 }

src/main/java/core/basesyntax/ColorSupplier.java

@@ -3,8 +3,9 @@
 import java.util.Random;
 
 public class ColorSupplier {
-    public String getRandomColor() {
-        int index = new Random().nextInt(Color.values().length);
-        return Color.values()[index].toString();
+    private Random random;
+    public Color getRandomColor() {
+        int index = random.nextInt(Color.values().length);
+        return Color.values()[index];
     }
 }

src/main/java/core/basesyntax/Lottery.java

@@ -3,14 +3,13 @@
 import java.util.Random;
 
 public class Lottery {
+    private Random random;
+    private static final int MAX_BALL_NUMBER = 100;
     ColorSupplier colorSupplier = new ColorSupplier();
     public Ball getRandomBall() {
-        Random random = new Random();
-        int number = random.nextInt(100);
-        String color = colorSupplier.getRandomColor();
-        Ball ball = new Ball();
-        ball.setColor(color);
-        ball.setNumber(number);
+        int number = random.nextInt(MAX_BALL_NUMBER);
+        Color color = colorSupplier.getRandomColor();
+        Ball ball = new Ball(color, number);
         return ball;
     }
 }