Add custom_context when initialising buttons section