Updated class snippet