Yes, I'll work as well, but I kept the class in case we need to override any methods in the future.