Added the Class that I meant to add in the prev commit