ok, moved the logic to context class