Actually adding test class now