diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..9f11b75
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+.idea/
diff --git a/action.yaml b/action.yaml
index e4a80cf..7a39bd6 100644
--- a/action.yaml
+++ b/action.yaml
@@ -4,8 +4,13 @@ author: 'Daniel Pacak'
 inputs:
   image-ref: # Docker image reference, e.g. alpine:3.10.2
     description: 'Docker image reference'
+    required: true
+  exit-code:
+    description: 'Exit code'
+    required: false
+    default: '0'
 runs:
   using: 'docker'
   image: 'docker://docker.io/aquasec/trivy:latest'
   args:
-    - ${{ inputs.image-ref }}
+    - --exit-code ${{ inputs.exit-code }} ${{ inputs.image-ref }}