added two classes which got missed