Add script to set state of nav