refactor: add improvements according to review.

README.md

@@ -85,23 +85,24 @@ export class MoviesStore extends ImmerComponentStore<MoviesState> {
 Provides an Immer-version of the `patchState` function from the `@ngrx/signals` package. In addition to partial state objects and updaters that update the state immutably, it accepts updater functions that update the state in a mutable manner. Similar to `patchState`, the `immerPatchState` function can be used to update the state of both SignalStore and SignalState.
 
 ```ts
-const UserState = signalStore(
-  withState({
-    id: 1,
-    name: { firstname: 'Konrad', lastname: 'Schultz' },
-    address: { city: 'Vienna', zip: '1010' },
-  }),
-  withComputed(({ name }) => ({
-    prettyName: computed(() => `${name.firstname()} ${name.lastname()}`),
-  }))
+const UserStore = signalStore(
+	withState({
+		user: { firstName: 'Konrad', lastName: 'Schultz' },
+		address: { city: 'Vienna', zip: '1010' },
+	}),
+	withMethods((store) => ({
+		setLastName(lastName: string): void {
+			immerPatchState(store, (state) => {
+				state.user.lastName = lastName;
+			});
+		},
+		setCity(city: string): void {
+			immerPatchState(store, (state) => {
+				state.address.city = city;
+			});
+		},
+	}))
 );
-
-const userState = new UserState();
-
-immerPatchState(userState, (state => {
-	state.name = { firstname: 'Lucy', lastname: 'Sanders' };
-	state.address.zip = '1020'
-}));
 ```
 
 Please note, that the updater function can only mutate a change without returning it or return an immutable 
@@ -112,13 +113,11 @@ This one is going to throw a runtime error:
 ```ts
 // will throw because of both returning and mutable change
 immerPatchState(userStore, (state) => {
-  state.name.lastname = 'Sanders'; // mutable change
-  return state; // returning state
+	state.name.lastname = 'Sanders'; // mutable change
+	return state; // returning state
 });
 ```
 
-
-
 ## `immerReducer`
 
 Inspired by [Alex Okrushko](https://twitter.com/alexokrushko), `immerReducer` is a reducer method that uses the Immer `produce` method.

src/signals/index.ts

@@ -3,7 +3,7 @@ import { immerReducer } from 'ngrx-immer';
 
 export type ImmerStateUpdater<State extends object> = (state: State) => void;
 
-function toFullStateUpdater<State extends object>(updater: PartialStateUpdater<State & {}> | MutableStateUpdater<State & {}>): (state: State) => State | void {
+function toFullStateUpdater<State extends object>(updater: PartialStateUpdater<State & {}> | ImmerStateUpdater<State & {}>): (state: State) => State | void {
 	return (state: State) => {
 		const patchedState = updater(state);
 		if (patchedState) {
@@ -13,7 +13,7 @@ function toFullStateUpdater<State extends object>(updater: PartialStateUpdater<S
 	};
 }
 
-export function immerPatchState<State extends object>(state: StateSignal<State>, ...updaters: Array<Partial<State & {}> | PartialStateUpdater<State & {}> | MutableStateUpdater<State & {}>>) {
+export function immerPatchState<State extends object>(state: StateSignal<State>, ...updaters: Array<Partial<State & {}> | PartialStateUpdater<State & {}> | ImmerStateUpdater<State & {}>>) {
 	const immerUpdaters = updaters.map(updater => {
 		if (typeof updater === 'function') {
 			return immerReducer(toFullStateUpdater(updater)) as unknown as PartialStateUpdater<State & {}>;