Adjusted to your structure

Changed my classify function to work with yours