Add logic to this behavior