MAINT: Updated the naming scheme from reconstruct to set_state