如何在BottomSheetDialog安卓系统的中上部添加关闭按钮？

BottomSheetDialog dialog = new BottomSheetDialog(context);
dialog.setContentView(R.layout.custom_layout);

<ImageButton
    android:id="@+id/close_button"
    android:layout_width="wrap_content"
    android:layout_height="wrap_content"
    android:src="@drawable/ic_close"
    android:background="@android:color/transparent"
    android:layout_gravity="end"
    android:layout_marginEnd="16dp"
    android:layout_marginTop="16dp"/>

ImageButton closeButton = dialog.findViewById(R.id.close_button);
closeButton.setOnClickListener(new View.OnClickListener() {
    @Override
    public void onClick(View v) {
        dialog.dismiss();
    }
});