Update solution for ex 10