Addressed code review comment, see #3.