The class should now works!