mini improvements and fixes