Add variation to median problem from lesson 2.