Trying the second problem

Adding helper classes