Removed a class i forgot to, this was intergrated into the test class