Unit Test

Please add class definition.