Updated class name to mach new namespace so the code is cleaner