Updating log in/sign up screen and arranging code