Update Lab 01 TDD according to comments from Kirill