Fixes #65 bugs