fixing small minor issues