Fixing login button feature