Add basic toggle functionality