Optionalize rectangle augmentation.
Clean up code, add comments.
Make step1 a class (no more global previous timestep variable.)
Move shared memory allocation for mid and post processing images to downstream stages.