Added small assignment from a course in TDD