minor changes: restore edit functionality