Fixing bug and remove couts