updated small bugs