Actually adding test class >_<