Add flash & session classes