Get code toggle to work