Add class to session switch. Closes #7