Example solution given, not sure if everything works yet though