Added class for testing