Change the class name and add the write function