Set theme lite refactor
Add todo