fix bug and add line