Add import for Lock