1 files changed, 9 insertions, 11 deletions

diff --git a/baseline/source/audiobeam/audiobeam.c b/baseline/source/audiobeam/audiobeam.c
index 50ebfff..5f23daf 100644
--- a/baseline/source/audiobeam/audiobeam.c
+++ b/baseline/source/audiobeam/audiobeam.c
@@ -104,6 +104,8 @@ void audiobeam_init()
   unsigned int i;
   unsigned char *p;
   volatile char bitmask = 0;
+  // Reset heap
+  audiobeam_free_all();
 
   /*
     Apply volatile XOR-bitmask to entire input array.
@@ -577,17 +579,13 @@ void _Pragma( "entrypoint" ) audiobeam_main( void )
 
 
 int main( int argc, char **argv )
-{       
-        //SET_UP
-        int jobsComplete;
-        int maxJobs=100;
-        for (jobsComplete=0; jobsComplete<maxJobs; jobsComplete++){
-        //      START_LOOP
-                audiobeam_init();
-                audiobeam_main();
-        //      STOP_LOOP
-        }
-        //WRITE_TO_FILE
+{
+  SET_UP
+  for_each_job {
+    audiobeam_init();
+    audiobeam_main();
+  }
+  WRITE_TO_FILE
 
   return ( audiobeam_return() );
 }