Add expand behavior