some improvements

чуть подправил