Octopus Energy Interview Question

In the final interview I was asked how I would extend my solution to the coding exercise.