diff --git a/buildspec.yml b/buildspec.yml
index d091010..6f29dcb 100644
--- a/buildspec.yml
+++ b/buildspec.yml
@@ -44,10 +44,3 @@ phases:
     commands:
       - cd ${CODEBUILD_SRC_DIR}/${CODE_SRC_DIR}
       - echo "Checking for the AMI ID file"
-      - test -s ami_id.txt || exit 1
-
-artifacts:
-  files:
-    - '**/*'
-    discard-paths: yes
-    - ami_id.txt
\ No newline at end of file
diff --git a/rhel-arm-baseline.yaml b/rhel-arm-baseline.yaml
index 72b926a..9f15bca 100644
--- a/rhel-arm-baseline.yaml
+++ b/rhel-arm-baseline.yaml
@@ -2,31 +2,31 @@
 - name: RHEL ARM Baseline
   hosts: all
   become: true
+
   tasks:
     - name: Ensure latest packages are installed
-      package:
+      yum:
         name: "*"
         state: latest
 
     - name: Install required packages
-      package:
+      yum:
         name:
           - vim
           - git
           - wget
           - net-tools
           - unzip
-          - epel-release
           - htop
         state: present
 
     - name: Remove curl-minimal if installed
-      package:
+      yum:
         name: curl-minimal
         state: absent
 
     - name: Install curl
-      package:
+      yum:
         name: curl
         state: present
 
@@ -36,7 +36,7 @@
         state: present
         groups: wheel
         shell: /bin/bash
-        password: "{{ 'password' | password_hash('sha512') }}" # Change 'password' to a secure password or hash
+        password: "{{ 'secure_password' | password_hash('sha512') }}" # Replace 'secure_password' with a strong password
 
     - name: Configure SSH for exampleuser
       authorized_key:
@@ -45,7 +45,7 @@
         key: "{{ lookup('file', '/path/to/exampleuser_ssh_key.pub') }}" # Replace with the actual path to the public key
 
     - name: Ensure firewalld is installed and running
-      package:
+      yum:
         name: firewalld
         state: present