Small polish changes