Need those few changes

Thank you for the updates