Playing around with Lenses. Possibly make code look nicer later on.