Changes needed to compile and run correctly.
Removed duplicated test class