Class missing for some reason. Readded it