fixed som bugs