diff --git a/celeste/ffmpeg.sh b/celeste/ffmpeg.sh
index eb69d29..24f6214 100644
--- a/celeste/ffmpeg.sh
+++ b/celeste/ffmpeg.sh
@@ -22,8 +22,6 @@ render_dir () {
 		$OUTPUT_DIR/${1##*/}.mp4
 }
 
-
-
 # Todo: error out if exists
 mkdir -p $OUTPUT_DIR
 
@@ -50,17 +48,18 @@ ffmpeg \
 	-safe 0 \
 	-i video_merge_list \
 	-vf "scale=1024x1024:flags=neighbor" \
-	$OUTPUT_DIR/00-all.mp4
+	$SC_ROOT/1x.mp4
 
 rm video_merge_list
 
 # Make accelerated video
 ffmpeg \
 	-loglevel error -stats -y \
-	-i $OUTPUT_DIR/00-all.mp4 \
+	-i $SC_ROOT/1x.mp4 \
 	-framerate 60 \
 	-filter:v "setpts=0.125*PTS" \
 	$SC_ROOT/8x.mp4
 
 echo "Cleaning up..."
+
 rm -dr $OUTPUT_DIR
diff --git a/polecart/basic/main.py b/polecart/basic/main.py
index 62c2b5d..614e463 100755
--- a/polecart/basic/main.py
+++ b/polecart/basic/main.py
@@ -200,13 +200,12 @@ for ep in range(num_episodes):
 
 
 
-
+		state = next_state
+		
+		
 		# Only train the network if we have enough
 		# transitions in memory to do so.
 		if len(memory) >= BATCH_SIZE:
-
-			state = next_state
-
 			# Run optimizer
 			optimize.optimize_model(
 				memory,