Adjust context menu