cli: strange exception on arrange

[hachreak]

On my computer fails on arrange the dataset:

15:05 $ maskrcnn-modanet datasets arrange
/home/hachreak/.maskrcnn-modanet/savedvars.json
Traceback (most recent call last):
  File "/tmp/maskrcnn-modanet/bin/maskrcnn-modanet", line 11, in <module>
    load_entry_point('maskrcnn-modanet', 'console_scripts', 'maskrcnn-modanet')()
  File "/tmp/maskrcnn-modanet/local/lib/python2.7/site-packages/click/core.py", line 764, in __call__
    return self.main(*args, **kwargs)
  File "/tmp/maskrcnn-modanet/local/lib/python2.7/site-packages/click/core.py", line 717, in main
    rv = self.invoke(ctx)
  File "/tmp/maskrcnn-modanet/local/lib/python2.7/site-packages/click/core.py", line 1137, in invoke
    return _process_result(sub_ctx.command.invoke(sub_ctx))
  File "/tmp/maskrcnn-modanet/local/lib/python2.7/site-packages/click/core.py", line 1137, in invoke
    return _process_result(sub_ctx.command.invoke(sub_ctx))
  File "/tmp/maskrcnn-modanet/local/lib/python2.7/site-packages/click/core.py", line 956, in invoke
    return ctx.invoke(self.callback, **ctx.params)
  File "/tmp/maskrcnn-modanet/local/lib/python2.7/site-packages/click/core.py", line 555, in invoke
    return callback(*args, **kwargs)
  File "/tmp/maskrcnn-modanet/maskrcnn_modanet/cli/main.py", line 89, in arrange
    from .. import arrange_images
  File "/tmp/maskrcnn-modanet/maskrcnn_modanet/arrange_images.py", line 14, in <module>
    from IPython.display import display
ImportError: No module named IPython.display

Maybe because the import, but It looks like you are not using it.
Please, could you clean the code from everything you are not using (especially the imports).
Thanks