Add function to identify the login screen