fix bugs
fix ouput bug