fixed som e minor stuff