Add display toggle script