Major updates to the shiny app