Add theme check