diff --git a/Dockerfile b/Dockerfile
index f993dc4997480bca05d7ad6f349e5acb73896006..2cece893f5a375acd22dae382f13d8d091a4f9d3 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -251,7 +251,7 @@ RUN wget -q -O /WSRT_Measures.ztar \
     rm /WSRT_Measures.ztar
 
 # Some python stuff
-RUN python3 -m pip install h5py pandas pyyaml astropy matplotlib scipy shapely bdsf ipython
+RUN python3 -m pip install h5py pandas pyyaml astropy matplotlib scipy shapely bdsf ipython radio_beam scikit-learn
 #    cd /src && \
 #   git clone https://github.com/lofar-astron/PyBDSF.git && \
 #  cd /src/PyBDSF && \
diff --git a/imcal.py b/imcal.py
index 7f57c3fe1fe814eba300b0ebf723098266e2cc20..64bc629f249e2a5bf51c590c693c2f63bc8d7ba2 100755
--- a/imcal.py
+++ b/imcal.py
@@ -9,7 +9,7 @@ import sys
 import matplotlib
 matplotlib.use('agg')
 import matplotlib.pyplot as plt
-from matplotlib.patches import Circle
+# from matplotlib.patches import Circle
 import numpy as np
 import subprocess
 from subprocess import Popen as Process, TimeoutExpired, PIPE
@@ -21,8 +21,8 @@ import logging
 import yaml
 import argparse
 
-from astropy.coordinates import SkyCoord
-from astropy.time import Time
+# from astropy.coordinates import SkyCoord
+# from astropy.time import Time
 import astropy.units as u
 from astropy.io import fits
 
@@ -666,7 +666,7 @@ if __name__ == "__main__":
     logging.info('Starting logger for {}'.format(__name__))
     logger = logging.getLogger(__name__)
 
-    t0 = Time.now()
+    # t0 = Time.now()
     parser = argparse.ArgumentParser(description='DDCal Inputs')
     parser.add_argument('msin', help='MS file to process')
     parser.add_argument('-c', '--config', action='store',
@@ -680,6 +680,6 @@ if __name__ == "__main__":
         os.path.join(os.path.dirname(os.path.realpath(__file__)), 'imcal.yml')
     # msin = args.msin
     main(args.msin, outbase=args.outbase, steps=args.steps, cfgfile=configfile)
-    extime = Time.now() - t0
-    print("Execution time: {:.1f} min".format(extime.to("minute").value))
+    # extime = Time.now() - t0
+    # print("Execution time: {:.1f} min".format(extime.to("minute").value))