Added in the leave/stay functionality.