adding function so train model knows that the lights turned on