Changed functionality to radio buttons