Add login, fix bugs