Fixes #7

Yup, adds it in!