Closes #9 with fixing the bug.