added tests for checking custom class