Added necessary methods to use with toggle button
see previous