Fix the warning and error:

[alihasanahmedkhan]

FutureWarning: In a future version of pandas all arguments of DataFrame.drop except for the argument 'labels' will be keyword-only

Fix the error:
NameError: name 'display' is not defined

Below were the complete Warning and Error:

FutureWarning: In a future version of pandas all arguments of DataFrame.drop except for the argument 'labels' will be keyword-only
  self.X = self.train.drop(self.y,1)
Traceback (most recent call last):
  File "tuba.py", line 14, in <module>
    chart = mc.make_charts()
  File "/home/sahil/pywedge/pywedge.py", line 42, in make_charts
    display(header)
NameError: name 'display' is not defined

[taknev83]

Hi,

Thanks for the PR & for your time :) I am tied up in an another project, will merge this PR soon...

[alihasanahmedkhan]

Hi,
Thank you for your response. Sure no problem at all.