Added updated class file