fix the bugs.....