diff --git a/ArrayManipulation.java b/ArrayManipulation.java
index 3f2f6a8..8ae9f50 100644
--- a/ArrayManipulation.java
+++ b/ArrayManipulation.java
@@ -1,26 +1,27 @@
 
 import java.util.Arrays;
+
 public class ArrayManipulation {
   public static int[] reverseArray(int[] arr) {
-    for (int i = 0; i < arr.length/2; i++) {
-      int tmp = arr[i]; arr[i] = arr[arr.length - 1 - i];
+    for (int i = 0; i < arr.length / 2; i++) {
+      int tmp = arr[i];
+      arr[i] = arr[arr.length - 1 - i];
       arr[arr.length - 1 - i] = tmp;
     }
     return arr;
   }
 
-  public static int[] removeLast(int[] arr) {
-    int[] lastRemoved = new int [arr.length - 1];
-    for (int i = 0; i < lastRemoved.length; i = i + 1) {
-      lastRemoved[i] = arr[i];
+  public static int[] squareEach(int[] arr){
+    int[] square = new int[arr.length];
+    for (int i = 0; i < arr.length; i++) {
+      square[i] = arr[i] * arr[i];
     }
-    return lastRemoved;
+    return square;
   }
-
-      
+  
   public static void main(String[] args) {
-    int[] arr = {1,2,3,4,5};
+    int[] arr = { 1, 2, 3, 4, 5 };
     System.out.println(Arrays.toString(reverseArray(arr)));
-    System.out.println(Arrays.toString(removeLast(arr)));
-    }
-  }
\ No newline at end of file
+    System.out.println(Arrays.toString(squareEach(arr)));
+  }
+}