Fixes #2

Forgot to close issue #2, my bad.