Added a simpler solution

There is an easier way to go about this.