Stick to the following improvements 1. We want improvements on type hints 2. We want improvements on the README file