For getting class info from output.txt