Add shift key support