Adding context class