diff --git a/auton_survival/models/dsm/__init__.py b/auton_survival/models/dsm/__init__.py
index df8e1c3..492e11a 100644
--- a/auton_survival/models/dsm/__init__.py
+++ b/auton_survival/models/dsm/__init__.py
@@ -60,6 +60,9 @@
 Deep Recurrent Survival Machines
 --------------------------------
 
+<img width=100% src=https://ndownloader.figshare.com/files/28329918>
+
+
 **Deep Recurrent Survival Machines (DRSM)** builds on the original **DSM**
 model and allows for learning of representations of the input covariates using
 **Recurrent Neural Networks** like **LSTMs, GRUs**. Deep Recurrent Survival
diff --git a/auton_survival/models/dsm/dsm_torch.py b/auton_survival/models/dsm/dsm_torch.py
index 7255a10..93b5c10 100644
--- a/auton_survival/models/dsm/dsm_torch.py
+++ b/auton_survival/models/dsm/dsm_torch.py
@@ -91,7 +91,6 @@ def create_representation(inputdim, layers, activation, bias=False):
   return nn.Sequential(*modules)
 
 
-
 class DeepSurvivalMachinesTorch(torch.nn.Module):
   """A Torch implementation of Deep Survival Machines model.