Small improvements for IDE type hinting