My changes for Task 01

Glad to see this change here. It was a tricky one and a scenario you'll encounter many times.