While you make this function, use default arguments or something else different to the defines I used above to define the number of classes, dimension, etc.