work on class validation