updated the problem by adding a class