Added context parameter