Added class info to the file