Login flow changes + toggle function for loading