Add set LED function