Add back login code