work on the mistakes