Add context class