Extra class that was missing