Added n_classes command line param
Got rid of all references to n_class coming from data, I think
Very very early stages of a Python 3 rewrite. (In a notebook for now)