Added tests for class extension.