Add initialization logic to main method of Display