Added another callback to component for finer-grained control