changed test file
added class