fixing small small changes