Added functions to control the display module