Adding functions to display the LED based on the classification