This is the class for challenge 2