Got display toggle working!