updated a class

updated a class