Added method get_component to trainable