Added context menu for selecting correct context class