Add theme handling