diff --git a/main.cpp b/main.cpp
index 96f4e5c..1a92ebc 100644
--- a/main.cpp
+++ b/main.cpp
@@ -22,9 +22,12 @@
 #include "CUDAStream.h"
 #elif defined(OCL)
 #include "OCLStream.h"
+#elif defined(RAJA)
+#include "RAJAStream.hpp"
+#elif defined(KOKKOS)
+#include "KOKKOSStream.hpp"
 #endif
 
-
 unsigned int ARRAY_SIZE = 52428800;
 unsigned int num_times = 10;
 unsigned int deviceIndex = 0;
@@ -80,6 +83,14 @@ void run()
   // Use the OpenCL implementation
   stream = new OCLStream<T>(ARRAY_SIZE, deviceIndex);
 
+#elif defined(RAJA)
+  // Use the RAJA implementation
+  stream = new RAJAStream<T>(ARRAY_SIZE, deviceIndex);
+
+#elif defined(KOKKOS)
+  // Use the Kokkos implementation
+  stream = new KOKKOSStream<T>(ARRAY_SIZE, deviceIndex);
+
 #endif
 
   stream->write_arrays(a, b, c);