Override default classes with our own