Add functionality to themes