Add test class, still have bug when calling cache