yes, this is a better way. I did it because, I copied the previous class and change the behavior.