Making the UI similar to the login screen