diff --git a/7_lstm.py b/7_lstm.py
index ebfb7a9..0f0f93d 100644
--- a/7_lstm.py
+++ b/7_lstm.py
@@ -45,7 +45,7 @@ def model(X, W, B, init_state, lstm_size):
     lstm = rnn_cell.BasicLSTMCell(lstm_size, forget_bias=1.0)
 
     # Get lstm cell output, time_step_size (28) arrays with lstm_size output: (batch_size, lstm_size)
-    outputs, states = rnn.rnn(lstm, X_split, initial_state=init_state)
+    outputs, _states = rnn.rnn(lstm, X_split, initial_state=init_state)
 
     # Linear activation
     # Get the last output
@@ -82,7 +82,7 @@ def model(X, W, B, init_state, lstm_size):
             sess.run(train_op, feed_dict={X: trX[start:end], Y: trY[start:end],
                                           init_state: np.zeros((batch_size, state_size))})
 
-        test_indices = np.arange(len(teX))  # Get A Test Batch
+        test_indices = np.arange(len(teX))  # Get a test batch
         np.random.shuffle(test_indices)
         test_indices = test_indices[0:test_size]