Add login control