ADDED NEW CLASS