fixed litte bugs