Adding in toggle