Fixes #5 + documentation updates