More class checks, we love oop