Add function to set screen brightness