Adding end of class relfection