Add option for different wakelock types (partial, full, dim screen)