Fixes issues #6 and #7