Adding case 4 to switch