Fixes to Test Main 2. (Doesn't Capture Images)