Added a few lines in the component class.