Add code for login button and window