add new class
to the way of assignm 2
Business logic