From 730d2c8255bb1cb38f9f76aa528a2c033d8a0db9 Mon Sep 17 00:00:00 2001
From: Wolfgang Kerzendorf <wkerzendorf@gmail.com>
Date: Sat, 24 Oct 2015 15:16:57 +0200
Subject: [PATCH] some small fixes

---
 tardis/simulation/base.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/tardis/simulation/base.py b/tardis/simulation/base.py
index bfa0c31b9b8..303337e6e76 100644
--- a/tardis/simulation/base.py
+++ b/tardis/simulation/base.py
@@ -10,9 +10,6 @@
 
 from astropy import units as u
 
-from collections import OrderedDict
-
-from tardis.model import Radial1DModel
 from tardis.montecarlo.base import MontecarloRunner
 
 # Adding logging support
@@ -145,7 +142,8 @@ def log_plasma_state(self, t_rad, w, t_inner, next_t_rad, next_w,
             next t_rad
         next_w: ~astropy.units.Quanity
             next_w
-        log_sampling
+        log_sampling: ~int
+            the n-th shells to be plotted
 
         Returns
         -------