Octopus Energy interview question

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