Bug fixes.

Fixes #30 & #31