Fixes #9 and #10

Hopefully