Python code is completed, There is a bug to be fixed