fixes #4 homework