Added missing newly added class