Add login handling